"""
安装依赖 pip install mcp
"""
from mcp.server.fastmcp import FastMCP

mcp = FastMCP("mcpdemo")


@mcp.tool()
def add(a: int, b: int) -> int:
    """计算两个数字的和
    Args:
        a: 第一个数字
        b: 第二个数字
    """
    print(f"计算 {a} + {b} = add({a}, {b})")
    return a + b


@mcp.tool()
def get_weather(city: str) -> str:
    """获取某个城市的天气
    Args:
        city: 具体城市
    """
    print(f"获取 {city} 的天气")
    return f"{city}天气不错"


@mcp.resource("greeting://{name}")
def greeting(name: str) -> str:
    """
    生成欢迎语
    """
    print(f"生成欢迎语: greeting://{name}")
    return f"欢迎 {name}"


if __name__ == "__main__":
    # 以 sse 协议暴露服务
    # mcp.run(transport='sse')
    # 以 stdio 协议暴露服务
    mcp.run(transport='stdio')
